#include <db.h> 
#include <stdio.h>
#include <stdlib.h>
#include <pthread.h>

/* DB的函数执行完成后，返回0代表成功，否则失败 */
void print_error(int ret)
{
	if(ret != 0)
		printf("ERROR: %s\n",db_strerror(ret));
}

/* 数据结构DBT在使用前，应首先初始化，否则编译可通过但运行时报参数错误  */
void init_DBT(DBT * key, DBT * data)
{
	memset(key, 0, sizeof(DBT));
	memset(data, 0, sizeof(DBT));
}

void main(void)
{
	DB *dbp;           
	DBT key, data;
	u_int32_t flags;  
	int ret;

	char *fruit = "apple";
	int number = 15;
	
	typedef struct customer
	{
		int  c_id;
		char name[10];
		char address[20];
		int  age;
	} CUSTOMER;
	CUSTOMER cust; 
	int key_cust_c_id = 1;

	cust.c_id = 1;
	strncpy(cust.name, "javer", 9);
	strncpy(cust.address, "chengdu", 19);
	cust.age = 32;

	/* 首先创建数据库句柄 */
	ret = db_create(&dbp, NULL, 0);
	print_error(ret);
	
	/* 创建数据库标志 */
	flags = DB_CREATE;    

	/* 创建一个名为single.db的数据库，使用B+树访问算法，本段代码演示对简单数据类型的处理 */
	ret = dbp->open(dbp, NULL, "single.db", NULL, DB_BTREE, flags, 0); 
	print_error(ret);

	init_DBT(&key, &data);
	
	/* 分别对关键字和数据赋值和规定长度 */
	key.data = fruit;
	key.size = strlen(fruit) + 1;
	data.data = &number;
	data.size = sizeof(int);

	/* 把记录写入数据库中，不允许覆盖关键字相同的记录 */
	ret = dbp->put(dbp, NULL, &key, &data,DB_NOOVERWRITE); 
	print_error(ret);
	
/* 手动把缓存中的数据刷新到硬盘文件中，实际上在关闭数据库时，数据会被自动刷新 */
//dbp->sync()；
	
init_DBT(&key, &data);

	key.data = fruit;
	key.size = strlen(fruit) + 1;
	
	/* 从数据库中查询关键字为apple的记录 */
	ret = dbp->get(dbp, NULL, &key, &data, 0);
	print_error(ret);

	/* 特别要注意数据结构DBT的字段data为void *型，所以在对data赋值和取值时，要做必要的类型转换。 */
	printf("The number = %d\n", *(int*)(data.data));
	
	if(dbp != NULL)
    		dbp->close(dbp, 0); 

	ret = db_create(&dbp, NULL, 0);
	print_error(ret);

	flags = DB_CREATE;    

	/* 创建一个名为complex.db的数据库，使用HASH访问算法，本段代码演示对复杂数据结构的处理 */
	ret = dbp->open(dbp, NULL, "complex.db", NULL, DB_HASH, flags, 0); 
	print_error(ret);

	init_DBT(&key, &data);

	key.size = sizeof(int);
	key.data = &(cust.c_id);

	data.size = sizeof(CUSTOMER);
	data.data = &cust;

	ret = dbp->put(dbp, NULL, &key, &data,DB_NOOVERWRITE);
	print_error(ret);
	
	memset(&cust, 0, sizeof(CUSTOMER));
	
	key.size = sizeof(int);
	key.data = &key_cust_c_id;

	data.data = &cust;
	data.ulen = sizeof(CUSTOMER); 
	data.flags = DB_DBT_USERMEM;

	dbp->get(dbp, NULL, &key, &data, 0);
	print_error(ret);
	
	printf("c_id = %d name = %s address = %s age = %d\n", 
		cust.c_id, cust.name, cust.address, cust.age);
	
	if(dbp != NULL)
    		dbp->close(dbp, 0); 
}


